package com.heima.leetcode.practice;

import java.util.HashMap;
import java.util.Map;

/**
 * leetcode 106. 从中序与后序遍历序列构造二叉树
 *
 * @author 勾新杰
 * @version 1.0
 * @date 2025/1/11 12:27
 */
public class E106 {

    /**
     * 存放中序遍历的节点值和索引
     */
    private final Map<Integer, Integer> map = new HashMap<>();

    /**
     * 递归构建树
     *
     * @param inorder   中序遍历结果
     * @param postorder 后序遍历结果
     * @return 根节点
     */
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        for (int i = 0; i < inorder.length; i++) map.put(inorder[i], i);
        return buildTree(
                postorder,
                0,
                postorder.length - 1,
                inorder,
                0,
                inorder.length - 1);
    }

    /**
     * 递归构建树
     *
     * @param postorder 后序遍历结果
     * @param postBegin 后序遍历结果起始索引
     * @param postEnd   后序遍历结果结束索引
     * @param inorder   中序遍历结果
     * @param inBegin   中序遍历结果起始索引
     * @param inEnd     中序遍历结果结束索引
     * @return 根节点
     */
    private TreeNode buildTree(
            int[] postorder,
            int postBegin,
            int postEnd,
            int[] inorder,
            int inBegin,
            int inEnd) {
        // 1. 递归结束条件：当前树遍历结果为空（当前树前序遍历和中序遍历对应的）
        if (postBegin > postEnd) return null;
        // 2. 找到当前树的根节点的值
        int rootValue = postorder[postEnd];
        Integer median = map.get(rootValue);
        int leftLength = median - inBegin;
        TreeNode root = new TreeNode(rootValue);
        // 3. 构建左子树、右子树
        root.left = buildTree(
                postorder,
                postBegin,
                postBegin + leftLength - 1,
                inorder,
                inBegin,
                median - 1
        );
        root.right = buildTree(
                postorder,
                postBegin + leftLength,
                postEnd - 1,
                inorder,
                median + 1,
                inEnd
        );
        // 4. 返回当前树
        return root;
    }
}
